1955C - Inhabitant of the Deep Sea - CodeForces Solution

from typing import List

class Solution:

    def compute(self, a: List[int], n: int, k: int) -> int:

        front = (k + 1) // 2
        back = (k // 2)
        for i in range(n):
            if (a[i] <= front):
                front -= a[i]
                a[i] = 0
            else:
                a[i] -= front
                break

        for i in range(n - 1, -1, -1):
            if (a[i] <= back):
                back -= a[i]
                a[i] = 0
            else:
                a[i] -= back
                break

        return a.count(0)


if __name__ == '__main__':
    cl = input()
    sln = Solution()

    for i in range(int(cl)):
        cl1 = input()
        a1 = cl1.split(' ')

        cl2 = input()
        a2 = cl2.split(' ')
        print(sln.compute([int(p) for p in a2], int(a1[0]), int(a1[1])))